/**
* Copyright (C) 2013-2024 Nanjing Pengyun Network Technology Co., Ltd.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
*     http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/ 

package py.test;

import static org.junit.Assert.assertTrue;

import java.util.TreeSet;
import org.junit.Test;

public class TreeSetTest extends TestBase {
  public TreeSetTest() throws Exception {
    super.init();
  }

  @Test
  public void add() throws Exception {
    TreeSet<Person> sortContexts = new TreeSet<Person>();
    assertTrue(sortContexts.add(new Person("test1", 1)));
    assertTrue(!sortContexts.add(new Person("test2", 1)));
  }

  class Person implements Comparable<Person> {
    public int age;
    public String name;

    public Person(String name, int age) {
      this.age = age;
      this.name = name;
    }

    @Override
    public int compareTo(Person o) {
      if (age > o.age) {
        return 1;
      } else if (age == o.age) {
        return 0;
      } else {
        return -1;
      }
    }

    public boolean equals(Object obj) {
      if (obj instanceof Person) {
        Person r = (Person) obj;
        if (this == r) {
          return true;
        }
      }
      return false;
    }
  }
}
